from langchain_community.chat_models import ChatTongyi

# 调用线上大模型

# api-key sk-bd4f583e4594462da8787d6a60b794ac

# llm = ChatTongyi(
#     model='qwen-long',
#     temperature=0.1,
#     top_7=0.7,
#     api_key='sk-bd4f583e4594462da8787d6a60b794ac'
# )
# print(llm.invoke('什么是DashScope'))

# 调用本地大模型
'''
    from langchain_community import ollama 已经过时了 需要重新安装
    pip install -U langchain-ollama
'''
from langchain_ollama import OllamaLLM

llm = OllamaLLM(model='qwen2.5:7b')
# print(llm.invoke('你叫什么名字'))

from langchain.schema import HumanMessage, AIMessage, SystemMessage

'''
HumanMessage:需要问的问题
AIMessage
SystemMessage:定义对话的背景或者大语言模型的身份
'''
system_message = SystemMessage(content='你是一个AI开发工程师')
human_message = HumanMessage(content='我想学习AI，不知道从什么地方开始学，请给出学习路线')
messages = [
    system_message,
    human_message
]
print(llm.invoke(messages))
